2023.08.02 NumPyからTensor変換【numpy, torch】

numpy ndarray から pytorch tensor へ変換するには、 numpyのうちに型をfloat32にキャストして、from_numpy()関数を使ってtensorに変換する。

code:python

x_numpy = np.random.randn(10)

x_numpy = x_numpy.astype(np.float32)

x_tensor = torch.from_numpy(x_numpy).clone()